package public

import (
	"path"

	"github.com/spf13/viper"
)

var (
	BaseConf, ProxyConf *viper.Viper
)

func InitConf(mode string) {
	BaseConf = viper.New()
	confPath := path.Join(".", "conf", mode)

	BaseConf.AddConfigPath(confPath)
	BaseConf.SetConfigName("base") // 配置的名称
	BaseConf.SetConfigType("toml") // 配置文件的类型
	err := BaseConf.ReadInConfig() // 读取所有文件
	if err != nil {
		panic(err)
	}

	ProxyConf = viper.New()
	ProxyConf.AddConfigPath(confPath)
	ProxyConf.SetConfigName("proxy") // 配置的名称
	ProxyConf.SetConfigType("toml")  // 配置文件的类型
	err = ProxyConf.ReadInConfig()   // 读取所有文件
	if err != nil {
		panic(err)
	}
}
